FaceDetectionModel

class FaceDetectionModel(internalModel: OnnxInferenceModel, modelKindDescription: String?) : FaceDetectionModelBase<BufferedImage> , InferenceModel
Content copied to clipboard

Face detection model implementation.

See also

Constructors

FaceDetectionModel
Link copied to clipboard
fun FaceDetectionModel(internalModel: OnnxInferenceModel, modelKindDescription: String? = null)
Content copied to clipboard

Functions

close
Link copied to clipboard
open override fun close()
Content copied to clipboard
convert
Link copied to clipboard
open override fun convert(output: OrtSession.Result): List<DetectedObject>
Content copied to clipboard

Converts raw model output to the result.

copy
Link copied to clipboard
open override fun copy(copiedModelName: String?, saveOptimizerState: Boolean, copyWeights: Boolean): InferenceModel
Content copied to clipboard
detectFaces
Link copied to clipboard
fun detectFaces(image: BufferedImage, topK: Int = 5, iouThreshold: Float = 0.5f): List<DetectedObject>
Content copied to clipboard

Detects topK faces on the given image. If topK is negative all detected faces are returned.

initializeWith
Link copied to clipboard
open override fun initializeWith(vararg executionProviders: ExecutionProvider)
Content copied to clipboard

Initialize the model with the specified executions providers.

predict
Link copied to clipboard
open override fun predict(inputData: FloatArray): Int
Content copied to clipboard
open fun predict(input: BufferedImage): List<DetectedObject>
Content copied to clipboard

Makes prediction on the given input.

predictSoftly
Link copied to clipboard
open override fun predictSoftly(inputData: FloatArray, predictionTensorName: String): FloatArray
Content copied to clipboard
reshape
Link copied to clipboard
open override fun reshape(vararg dims: Long)
Content copied to clipboard
summary
Link copied to clipboard
open override fun summary(): ModelSummary
Content copied to clipboard

Properties

inputDimensions
Link copied to clipboard
open override val inputDimensions: LongArray
Content copied to clipboard
internalModel
Link copied to clipboard
open override val internalModel: OnnxInferenceModel
Content copied to clipboard

Model used to make predictions.

modelKindDescription
Link copied to clipboard
open override val modelKindDescription: String?
Content copied to clipboard

High-level description of the model. Used for model summary printing. For the model hub models it equals to the string representation of the OnnxModelType

preprocessing
Link copied to clipboard
open override val preprocessing: Operation<BufferedImage, Pair<FloatArray, TensorShape>>
Content copied to clipboard

Preprocessing operation specific to this model.